package cn.xuchunh.sword;

/**
 * 给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。
 * 注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
 * Created by XuChunH on 2016/9/18.
 */
public class GetNext {

    public TreeLinkNode solution(TreeLinkNode pNode){
        if(pNode == null) {
            return null;
        }
        if(pNode.right != null){
            TreeLinkNode node = pNode.right;
            while (node.left != null){
                node = node.left;
            }
            return node;
        }
        while (pNode.next != null) {
            if(pNode == pNode.next.left) {
                return pNode.next;
            }
            pNode = pNode.next;
        }
        return null;
    }

}
